﻿// --------------------------------------------------------------------------------------------------------------------
// <copyright file="DataReaderTransferTaskHelper.cs" company="DHGMS Solutions">
//   Copyright 2004-2014 DHGMS Solutions.
//      
//   Licensed under the Apache License, Version 2.0 (the "License");
//   you may not use this file except in compliance with the License.
//   You may obtain a copy of the License at
//      
//   http://www.apache.org/licenses/LICENSE-2.0
//      
//   Unless required by applicable law or agreed to in writing, software
//   distributed under the License is distributed on an "AS IS" BASIS,
//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//   See the License for the specific language governing permissions and
//   limitations under the License.
// </copyright>
// <summary>
//   TODO: Update summary.
// </summary>
// --------------------------------------------------------------------------------------------------------------------

using System.Data;
using System.Threading.Tasks;

namespace Dhgms.TplHelper.Model.TaskHelper
{
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text;

    using Dhgms.DataManager.Model.Helper.Database;
    using Dhgms.TplHelper.Model.DataReaderConsumer;
    using Dhgms.TplHelper.Model.DataReaderProducer;
    using Dhgms.TplHelper.Model.Info;
    using Dhgms.TplHelper.Model.Info.TaskResult;

    /// <summary>
    /// TODO: Update summary.
    /// </summary>
    public abstract class DataReaderTransferTaskHelper<TDataReaderProducer, TDataReaderConsumer, TTaskResultClass> : ReportableCountTaskHelper<TTaskResultClass, DataRetrievalProcessorResult<TTaskResultClass>>
        where TDataReaderProducer : IDataReaderProducer<TTaskResultClass>
        where TDataReaderConsumer : IDataReaderConsumer<TTaskResultClass>
        where TTaskResultClass : TaskResultBase, IReportableCount, new()
    {
        /// <summary>Initializes a new instance of the <see cref="DataReaderTransferTaskHelper{TDataReaderProducer,TDataReaderConsumer,TTaskResultClass}"/> class.</summary>
        /// <param name="name">The name.</param>
        protected DataReaderTransferTaskHelper(string name)
            : base(name)
        {
        }

        protected async override Task<DataRetrievalProcessorResult<TTaskResultClass>> GetProcess()
        {
            var producer = GetProducer();
            var consumer = GetConsumer();

            return await producer.ProduceAndConsume(dataReader => consumer.DataReaderAction(dataReader));
        }

        /// <summary>
        /// Gets the producer.
        /// </summary>
        /// <returns>The producer</returns>
        protected abstract TDataReaderProducer GetProducer();

        /// <summary>
        /// Gets the consumer.
        /// </summary>
        /// <returns>The consumer</returns>
        protected abstract TDataReaderConsumer GetConsumer();
    }
}
